pandas add flag if value is above certain value

30

# np.where function works as follows:
import numpy as np

# E.g. 1 - Set column values based on if another column is greater than or equal to 50
df['X'] = np.where(df['Y'] >= 50, 'yes', 'no')

# E.g. 2 - Replace values over 20000 with 0, otherwise keep original value
df['my_value'] = np.where(df.my_value > 20000, 0, df.my_value)

Comments

Submit
0 Comments